
#define NOGMM
#define NOMTL
#define EIGEN_GOOGLEHASH_SUPPORT 1

#include <ext/hash_map>
#include <google/dense_hash_map>
#include <google/sparse_hash_map>
#include <map>

#ifndef SIZE
#define SIZE 10000
#endif

#ifndef DENSITY
#define DENSITY 0.01
#endif

#ifndef REPEAT
#define REPEAT 1
#endif

#include "BenchSparseUtil.h"

#ifndef MINDENSITY
#define MINDENSITY 0.0004
#endif

#ifndef NBTRIES
#define NBTRIES 10
#endif

#define BENCH(X)                                                                                                       \
	timer.reset();                                                                                                     \
	for (int _j = 0; _j < NBTRIES; ++_j) {                                                                             \
		timer.start();                                                                                                 \
		for (int _k = 0; _k < REPEAT; ++_k) {                                                                          \
			X                                                                                                          \
		}                                                                                                              \
		timer.stop();                                                                                                  \
	}

static double rtime;
static double nentries;

template<typename SetterType>
void
dostuff(const char* name, EigenSparseMatrix& sm1)
{
	int rows = sm1.rows();
	int cols = sm1.cols();
	sm1.setZero();
	BenchTimer t;
	SetterType* set1 = new SetterType(sm1);
	t.reset();
	t.start();
	for (int k = 0; k < nentries; ++k)
		(*set1)(internal::random<int>(0, rows - 1), internal::random<int>(0, cols - 1)) += 1;
	t.stop();
	std::cout << "std::map =>      \t" << t.value() - rtime << " nnz=" << set1->nonZeros() << std::flush;

	// getchar();

	t.reset();
	t.start();
	delete set1;
	t.stop();
	std::cout << "  back: \t" << t.value() << "\n";
}

int
main(int argc, char* argv[])
{
	int rows = SIZE;
	int cols = SIZE;
	float density = DENSITY;

	EigenSparseMatrix sm1(rows, cols), sm2(rows, cols);

	nentries = rows * cols * density;
	std::cout << "n = " << nentries << "\n";
	int dummy;
	BenchTimer t;

	t.reset();
	t.start();
	for (int k = 0; k < nentries; ++k)
		dummy = internal::random<int>(0, rows - 1) + internal::random<int>(0, cols - 1);
	t.stop();
	rtime = t.value();
	std::cout << "rtime = " << rtime << " (" << dummy << ")\n\n";
	const int Bits = 6;
	for (;;) {
		dostuff<RandomSetter<EigenSparseMatrix, StdMapTraits, Bits>>("std::map     ", sm1);
		dostuff<RandomSetter<EigenSparseMatrix, GnuHashMapTraits, Bits>>("gnu::hash_map", sm1);
		dostuff<RandomSetter<EigenSparseMatrix, GoogleDenseHashMapTraits, Bits>>("google::dense", sm1);
		dostuff<RandomSetter<EigenSparseMatrix, GoogleSparseHashMapTraits, Bits>>("google::sparse", sm1);

		//     {
		//       RandomSetter<EigenSparseMatrix,GnuHashMapTraits,Bits> set1(sm1);
		//       t.reset(); t.start();
		//       for (int k=0; k<n; ++k)
		//         set1(internal::random<int>(0,rows-1),internal::random<int>(0,cols-1)) += 1;
		//       t.stop();
		//       std::cout << "gnu::hash_map => \t" << t.value()-rtime
		//                 << " nnz=" << set1.nonZeros() << "\n";getchar();
		//     }
		//     {
		//       RandomSetter<EigenSparseMatrix,GoogleDenseHashMapTraits,Bits> set1(sm1);
		//       t.reset(); t.start();
		//       for (int k=0; k<n; ++k)
		//         set1(internal::random<int>(0,rows-1),internal::random<int>(0,cols-1)) += 1;
		//       t.stop();
		//       std::cout << "google::dense => \t" << t.value()-rtime
		//                 << " nnz=" << set1.nonZeros() << "\n";getchar();
		//     }
		//     {
		//       RandomSetter<EigenSparseMatrix,GoogleSparseHashMapTraits,Bits> set1(sm1);
		//       t.reset(); t.start();
		//       for (int k=0; k<n; ++k)
		//         set1(internal::random<int>(0,rows-1),internal::random<int>(0,cols-1)) += 1;
		//       t.stop();
		//       std::cout << "google::sparse => \t" << t.value()-rtime
		//                 << " nnz=" << set1.nonZeros() << "\n";getchar();
		//     }
		std::cout << "\n\n";
	}

	return 0;
}
